package com.atguigu.daijia.order.testLock;

import cn.hutool.core.util.StrUtil;
import org.apache.commons.lang3.StringUtils;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;

import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * @Description:
 * @Author: wangshuo
 * @Date: 2024/8/3 17:09
 */
@Service
public class TestServiceImpl implements TestService {

    //    Lock lock = new ReentrantLock();
    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    @Autowired
    private RedissonClient redissonClient;

//    @Override
//      本地锁
//    public synchronized void testLock() {
//        lock.lock();
//        try {
//            //获取本锁保护的资源
//            System.out.println(Thread.currentThread().getName() + "开始执行任务。。。");
//            // 查询Redis中的num值
//            String value = this.stringRedisTemplate.opsForValue().get("num");
//            // 没有该值return
//            if (StringUtils.isBlank(value)) {
//                return;
//            }
//            // 有值就转成成int
//            int num = Integer.parseInt(value);
//            // 把Redis中的num值+1
//            this.stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));
//        } finally {
//            lock.unlock();
//        }
//    }

    //redis里面的setnex 命令
//    public void testLock() {
//        System.out.println(Thread.currentThread().getName() + "开始执行任务。。。");
//        //0.先尝试获取锁 setnx key val
////        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent("lock", "lock");
//        Boolean flag = stringRedisTemplate
//                .opsForValue().setIfAbsent("lock", "lock", 10, TimeUnit.SECONDS);
//        if (Boolean.TRUE.equals(flag)) {
//            //获取锁成功，执行业务代码
//            //1.先从redis中通过key num获取值  key提前手动设置 num 初始值：0
//            String value = stringRedisTemplate.opsForValue().get("num");
//            //2.如果值为空则非法直接返回即可
//            if (StringUtils.isBlank(value)) {
//                return;
//            }
//            //3.对num值进行自增加一
//            int num = Integer.parseInt(value);
//            stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));
//
//            //4.将锁释放
//            stringRedisTemplate.delete("lock");
//
//        } else {
//            try {
//                Thread.sleep(100);
//                this.testLock();
//            } catch (InterruptedException e) {
//                e.printStackTrace();
//            }
//        }
//    }

    // uuid 防止误删
  /*  public void testLock() {
        System.out.println(Thread.currentThread().getName() + "开始执行任务。。。");
        String uuid = UUID.randomUUID().toString();
        //0.先尝试获取锁 setnx key val
//        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent("lock", "lock");
        Boolean flag = stringRedisTemplate
                .opsForValue().setIfAbsent("lock", uuid, 10, TimeUnit.SECONDS);
        if (Boolean.TRUE.equals(flag)) {
            //获取锁成功，执行业务代码
            //1.先从redis中通过key num获取值  key提前手动设置 num 初始值：0
            String value = stringRedisTemplate.opsForValue().get("num");
            //2.如果值为空则非法直接返回即可
            if (StringUtils.isBlank(value)) {
                return;
            }
            //3.对num值进行自增加一
            int num = Integer.parseInt(value);
            stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));

            //4.将锁释放
            String redisUUID = stringRedisTemplate.opsForValue().get("lock");
            if (Objects.equals(redisUUID, uuid)) {
                stringRedisTemplate.delete("lock");
            }
        } else {
            try {
                Thread.sleep(100);
                this.testLock();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }*/

    //使用lua脚本保证原子性
/*    public void testLock() {
        System.out.println(Thread.currentThread().getName() + "开始执行任务。。。");
        String uuid = UUID.randomUUID().toString();
        //0.先尝试获取锁 setnx key val
//        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent("lock", "lock");
        Boolean flag = stringRedisTemplate
                .opsForValue().setIfAbsent("lock", uuid, 10, TimeUnit.SECONDS);
        if (Boolean.TRUE.equals(flag)) {
            //获取锁成功，执行业务代码
            //1.先从redis中通过key num获取值  key提前手动设置 num 初始值：0
            String value = stringRedisTemplate.opsForValue().get("num");
            //2.如果值为空则非法直接返回即可
            if (StringUtils.isBlank(value)) {
                return;
            }
            //3.对num值进行自增加一
            int num = Integer.parseInt(value);
            stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));

            //4.将锁释放  采用lua脚本实现
//            String redisUUID = stringRedisTemplate.opsForValue().get("lock");
//            if (Objects.equals(redisUUID, uuid)) {
//                stringRedisTemplate.delete("lock");
//            }
            //4.1 先创建脚本对象 DefaultRedisScript泛型脚本语言返回值类型 Long 0：失败 1：成功
            DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
            //4.2设置脚本文本
            String script = "if redis.call(\"get\",KEYS[1]) == ARGV[1]\n" +
                    "then\n" +
                    "    return redis.call(\"del\",KEYS[1])\n" +
                    "else\n" +
                    "    return 0\n" +
                    "end";
            redisScript.setScriptText(script);
            //4.3 设置响应类型
            redisScript.setResultType(Long.class);
            stringRedisTemplate.execute(redisScript, List.of("lock"), uuid);
        } else {
            try {
                Thread.sleep(100);
                this.testLock();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }*/

    //使用Redisson实现分布式锁
    public void testLock() {
        System.out.println(Thread.currentThread().getName() + "开始执行任务。。。");

        //0.创建锁对象
        RLock lock = redissonClient.getLock("lock1");

        //0.1 尝试加锁
        //0.1.1 lock() 阻塞等待一直到获取锁,默认锁有效期30s
//        lock.lock();

        //0.1.2 lock(long l, TimeUnit timeUnit) 手动设置锁的过期时长
        lock.lock(10,TimeUnit.SECONDS);

        //0.1.3 tryLock(long l, long l1, TimeUnit timeUnit)
//        try {
//            //0.1.3 tryLock(long l, long l1, TimeUnit timeUnit)
//            // 第一个参数：尝试获取锁，如果获取锁失败则等待30秒，超过30秒不在等待，
//            // 第二个参数：如果30s内获取到了锁，则锁的过期时间为10s
//            boolean tryLock = lock.tryLock(30, 10, TimeUnit.SECONDS);
//        } catch (InterruptedException e) {
//            throw new RuntimeException(e);
//        }

        //1.先从redis中通过key num获取值  key提前手动设置 num 初始值：0
        String value = stringRedisTemplate.opsForValue().get("num");
        //2.如果值为空则非法直接返回即可
        if (StringUtils.isBlank(value)) {
            return;
        }
        //3.对num值进行自增加一
        int num = Integer.parseInt(value);
        stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));

        //4.将锁释放
        lock.unlock();
    }


}

